//
// Created by fazhehy on 2024/2/26.
//

#ifndef NRF24L01_H
#define NRF24L01_H

#include "stm32f1xx_hal.h"

#define NRF24L01_SCK_PORT       GPIOB
#define NRF24L01_SCK_PIN        GPIO_PIN_13

#define NRF24L01_MOSI_PORT      GPIOB
#define NRF24L01_MOSI_PIN       GPIO_PIN_15

#define NRF24L01_MISO_PORT      GPIOB
#define NRF24L01_MISO_PIN       GPIO_PIN_14

#define NRF24L01_CSN_PORT       GPIOB
#define NRF24L01_CSN_PIN        GPIO_PIN_12

#define NRF24L01_CE_PORT        GPIOB
#define NRF24L01_CE_PIN         GPIO_PIN_11

#define NRF24L01_IRQ_PORT       GPIOA
#define NRF24L01_IRQ_PIN        GPIO_PIN_8

#define NRF24L01_SCK_OUT_H()        HAL_GPIO_WritePin(NRF24L01_SCK_PORT, NRF24L01_SCK_PIN, GPIO_PIN_SET)
#define NRF24L01_SCK_OUT_L()        HAL_GPIO_WritePin(NRF24L01_SCK_PORT, NRF24L01_SCK_PIN, GPIO_PIN_RESET)
#define NRF24L01_MOSI_OUT_H()       HAL_GPIO_WritePin(NRF24L01_MOSI_PORT, NRF24L01_MOSI_PIN, GPIO_PIN_SET)
#define NRF24L01_MOSI_OUT_L()       HAL_GPIO_WritePin(NRF24L01_MOSI_PORT, NRF24L01_MOSI_PIN, GPIO_PIN_RESET)
#define NRF24L01_CSN_OUT_H()        HAL_GPIO_WritePin(NRF24L01_CSN_PORT, NRF24L01_CSN_PIN, GPIO_PIN_SET)
#define NRF24L01_CSN_OUT_L()        HAL_GPIO_WritePin(NRF24L01_CSN_PORT, NRF24L01_CSN_PIN, GPIO_PIN_RESET)
#define NRF24L01_CE_OUT_H()         HAL_GPIO_WritePin(NRF24L01_CE_PORT, NRF24L01_CE_PIN, GPIO_PIN_SET)
#define NRF24L01_CE_OUT_L()         HAL_GPIO_WritePin(NRF24L01_CE_PORT, NRF24L01_CE_PIN, GPIO_PIN_RESET)

#define NRF24L01_MISO_IN()          HAL_GPIO_ReadPin(NRF24L01_MISO_PORT, NRF24L01_MISO_PIN)
#define NRF24L01_IRQ_IN()           HAL_GPIO_ReadPin(NRF24L01_IRQ_PORT, NRF24L01_IRQ_PIN)

//NRF24L01寄存器操作命令
#define NRF24L01_READ_REG        0x00  //读配置寄存器,低5位为寄存器地址
#define NRF24L01_WRITE_REG       0x20  //写配置寄存器,低5位为寄存器地址
#define NRF24L01_RD_RX_PLOAD     0x61  //读RX有效数据,1~32字节
#define NRF24L01_WR_TX_PLOAD     0xA0  //写TX有效数据,1~32字节
#define NRF24L01_FLUSH_TX        0xE1  //清除TX FIFO寄存器.发射模式下用
#define NRF24L01_FLUSH_RX        0xE2  //清除RX FIFO寄存器.接收模式下用
#define NRF24L01_REUSE_TX_PL     0xE3  //重新使用上一包数据,CE为高,数据包被不断发送.
#define NRF24L01_NOP             0xFF  //空操作,可以用来读状态寄存器
//SPI(NRF24L01)寄存器地址
#define NRF24L01_CONFIG          0x00  //配置寄存器地址;bit0:1接收模式,0发射模式;bit1:电选择;bit2:CRC模式;bit3:CRC使能;
//bit4:中断MAX_RT(达到最大重发次数中断)使能;bit5:中断TX_DS使能;bit6:中断RX_DR使能
#define NRF24L01_EN_AA           0x01  //使能自动应答功能  bit0~5,对应通道0~5
#define NRF24L01_EN_RXADDR       0x02  //接收地址允许,bit0~5,对应通道0~5
#define NRF24L01_SETUP_AW        0x03  //设置地址宽度(所有数据通道):bit1,0:00,3字节;01,4字节;02,5字节;
#define NRF24L01_SETUP_RETR      0x04  //建立自动重发;bit3:0,自动重发计数器;bit7:4,自动重发延时 250*x+86us
#define NRF24L01_RF_CH           0x05  //RF通道,bit6:0,工作通道频率;
#define NRF24L01_RF_SETUP        0x06  //RF寄存器;bit3:传输速率(0:1Mbps,1:2Mbps);bit2:1,发射功率;bit0:低噪声放大器增益
#define NRF24L01_STATUS          0x07  //状态寄存器;bit0:TX FIFO满标志;bit3:1,接收数据通道号(最大:6);bit4,达到最多次重发
//bit5:数据发送完成中断;bit6:接收数据中断;
#define NRF24L01_MAX_TX  	0x10  //达到最大发送次数中断
#define NRF24L01_TX_OK   	0x20  //TX发送完成中断
#define NRF24L01_RX_OK   	0x40  //接收到数据中断

#define NRF24L01_OBSERVE_TX      0x08  //发送检测寄存器,bit7:4,数据包丢失计数器;bit3:0,重发计数器
#define NRF24L01_CD              0x09  //载波检测寄存器,bit0,载波检测;
#define NRF24L01_RX_ADDR_P0      0x0A  //数据通道0接收地址,最大长度5个字节,低字节在前
#define NRF24L01_RX_ADDR_P1      0x0B  //数据通道1接收地址,最大长度5个字节,低字节在前
#define NRF24L01_RX_ADDR_P2      0x0C  //数据通道2接收地址,最低字节可设置,高字节,必须同RX_ADDR_P1[39:8]相等;
#define NRF24L01_RX_ADDR_P3      0x0D  //数据通道3接收地址,最低字节可设置,高字节,必须同RX_ADDR_P1[39:8]相等;
#define NRF24L01_RX_ADDR_P4      0x0E  //数据通道4接收地址,最低字节可设置,高字节,必须同RX_ADDR_P1[39:8]相等;
#define NRF24L01_RX_ADDR_P5      0x0F  //数据通道5接收地址,最低字节可设置,高字节,必须同RX_ADDR_P1[39:8]相等;
#define NRF24L01_TX_ADDR         0x10  //发送地址(低字节在前),ShockBurstTM模式下,RX_ADDR_P0与此地址相等
#define NRF24L01_RX_PW_P0        0x11  //接收数据通道0有效数据宽度(1~32字节),设置为0则非法
#define NRF24L01_RX_PW_P1        0x12  //接收数据通道1有效数据宽度(1~32字节),设置为0则非法
#define NRF24L01_RX_PW_P2        0x13  //接收数据通道2有效数据宽度(1~32字节),设置为0则非法
#define NRF24L01_RX_PW_P3        0x14  //接收数据通道3有效数据宽度(1~32字节),设置为0则非法
#define NRF24L01_RX_PW_P4        0x15  //接收数据通道4有效数据宽度(1~32字节),设置为0则非法
#define NRF24L01_RX_PW_P5        0x16  //接收数据通道5有效数据宽度(1~32字节),设置为0则非法
#define NRF24L01_FIFO_STATUS     0x17  //FIFO状态寄存器;bit0,RX FIFO寄存器空标志;bit1,RX FIFO满标志;bit2,3,保留
//bit4,TX FIFO空标志;bit5,TX FIFO满标志;bit6,1,循环发送上一数据包.0,不循环;
#define SI24R1_DYNPD             0x1c
#define SI24R1_FEATURE           0x1d

#define NRF24L01_TX_ADR_WIDTH    5   //5字节的地址宽度
#define NRF24L01_RX_ADR_WIDTH    5   //5字节的地址宽度
#define NRF24L01_TX_PLOAD_WIDTH  32  //20字节的用户数据宽度
#define NRF24L01_RX_PLOAD_WIDTH  32  //20字节的用户数据宽度
#define NRF24L01_TX_MODE         0
#define NRF24L01_RX_MODE         1

extern uint8_t receive_buffer[32];
extern uint8_t receive_flag;
extern uint8_t transmit_buffer[32];
extern uint8_t transmit_flag;

void nrf24l01_init();
uint8_t nrf24l01_check();
uint8_t nrf24l01_send_buffer(uint8_t * buffer);
uint8_t nrf24l01_receive_buffer(uint8_t * buffer);
void nrf24l01_set_tx_mode();
void nrf24l01_set_rx_mode();
void nrf24l01_receive_callback(uint16_t GPIO_Pin);
void nrf24l01_transmit_callback();
uint8_t nrf24l01_send_data(const uint8_t * data, uint8_t len);

#endif //NRF24L01_H
